﻿#pragma once

#include "Ability/Group/EditorAbilityGroupViewModel.h"
#include "EditorEngineObjectAbilityGroupViewModel.m.h"

class RTexture2D;
class REditorEngineObjectAbilityGroupModel;

RCLASS()
class REditorEngineObjectAbilityGroupViewModel : public REditorAbilityGroupViewModel
{
    ROBJECT

public:

    void BindModel(TSharedObjectPtr<REditorAbilityModel> InModel) override;
    TSharedObjectPtr<RTexture2D> GetIcon() const override;
    TSharedObjectPtr<RTexture2D> GetHighlightIcon() const override;

protected:

    TWeakObjectPtr<REditorEngineObjectAbilityGroupModel> Model;
    TSharedObjectPtr<RTexture2D> Icon;
    TSharedObjectPtr<RTexture2D> HighlightIcon;

};